package com.Oct_Twenty;


import com.sun.source.doctree.SeeTree;

import java.util.HashSet;
import java.util.Set;

/**
 * 547. 省份数量
 * 有 n 个城市，其中一些彼此相连，另一些没有相连。如果城市 a 与城市 b 直接相连，且城市 b 与城市 c 直接相连，那么城市 a 与城市 c 间接相连。
 *
 * 省份 是一组直接或间接相连的城市，组内不含其他没有相连的城市。
 *
 * 给你一个 n x n 的矩阵 isConnected ，其中 isConnected[i][j] = 1 表示第 i 个城市和第 j 个城市直接相连，而 isConnected[i][j] = 0 表示二者不直接相连。
 *
 * 返回矩阵中 省份 的数量。
 *
 *  
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/number-of-provinces
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 *
 *
 * 2022/10/21 00:17
 */

public class Solution547 {


    /**
     * 直接dfs深度搜索
     * @param isConnected
     * @return
     */
    public int findCircleNum(int[][] isConnected) {
        boolean[] visited=new boolean[isConnected.length];
        int ans=0;
        for (int i = 0; i < isConnected.length; i++) {
            if (!visited[i])
            {
                dfs(isConnected,visited,i);
                ans++;
            }
        }
        return ans;
    }

    void dfs(int[][] grid, boolean[] visited,int i){
        for (int j = 0; j < grid.length; j++) {
            if (grid[i][j]==1&&!visited[j])
            {
                visited[j]=true;
                dfs(grid,visited,j);
            }
        }
    }

    public static void main(String[] args) {
        Solution547 solution547=new Solution547();
        System.out.println(solution547.findCircleNum(new int[][]{{1,0,0,1},{0,1,1,0},{0,1,1,1},{1,0,1,1}}));
    }

    /**
     * 并查集搜索
     */
    int[] father=new int[205];

    public int findCircleNum2(int[][] isConnected) {
        for (int i = 0; i < father.length; i++) {
            father[i]=i;
        }
        for (int i = 0; i < isConnected.length; i++) {
            for (int j = 0; j < isConnected.length; j++) {
                if (isConnected[i][j]==1) Join(i,j);
            }
        }
        Set<Integer> set=new HashSet<>();
        for (int i = 0; i < isConnected.length; i++) {
            set.add(find(i));
        }
        return set.size();
    }

    int find(int u){
        return u==father[u]?u:(u=find(father[u]));
    }

    void Join(int u,int v){
        u=find(u);
        v=find(v);
        if (u==v) return;
        father[v]=u;
    }


}
